updates to assignment 1