fix issue with assignment 1