Key Brexit legislation to return to Commons in early June