updated to assignment 3