modified file for question 2