Finished seperating data using numpy
Messing around a bit more with numpy