Worked on the python programming task
Added some comments