modified demo1.py file to hw2a.py for homework2
removed an error which lead to wrong calculations
Finished homework 2