Added required assignment code
Added comments