Added ball class to draw