Got rid of extra code
Add ball class
Refactor