Made changes to make the assignment better.