How can I create this layout?

I want to have a layout which in desktop has 2 columns on the left side there are some texts, fields and is a submit button, on the right side just some texts but I want on small screen to have just one column where the right column would go between the texts and submit button of the left side, In this example I can't make the button go right below the red section and not have all that space in between. Here is also the link to the sandbox https://codesandbox.io/s/peaceful-pare-jttq33?file=/src/styles.css