updated to assignment 2
updated to assignment 2