Added as2.py for second assignment