Add createRectangle function