Added code to draw a rectangle