Major improvements. now have class for easily plotting stuff
All other changes. Small stuff