minor changes to assignment 1