Adjusted according to Assignment 2 instructions.

Just for the sake of a good grade.