add function for drawing square