Assignment 3 - Final revision

Changed comments
Minor changes