update to assignment 3