Update the draw method properly