Finished first data wrangling. Export into X and Y for stage 1 and stage 2.
Export to python script