error with drawing shapes
refine the draw method