continued work on assignment 2 code