Create ass1problem2.py