fix ex1 assignment 5