finished assn3 - need to add comments still