changed names & added comment

Final for assignment 2