Fixed issue with assignment 1.