fixed assignment 5 #35