Ball class fixed

TODO: clear other classes