Updated UI to draw lines