Create ass1problem4.py