Final source for assignment 2 - One additional update

Added some additional comments to the source to make it more
understandable.