updates to assignment 2