Lab 2 with square