added draw function and draw_ball function