Assignment 4 with shapes and stuff