Improved UI of the drawing component