changed the layout to fit constraints of assignment 3