Minor changes to assignment 3