CW3 update: added some comments

- Added a few comments to skeleton code
- Still needed to remove print statements
- Still need to check answers though!!!!