The UK will “get control over our borders” as free movement to and from the EU ends, ministers will say.