added squares, main, and print func