Moved ball code to 'ball' class