vpreet final changes for the assignment