here is my updated submission for assignment 2