Code clean up
Updated draw method
Can now create spheres