quick changes to assignment 3