Added a method for drawing a box