update to assignment 2