Added code and comments to file (completed assignment)