Fixed assignment 03
Added comments