file modified for the assignment 2