Added Required Functionality

Created a subroutine to draw the square and then handled the call to
that in main.