small changes to draw