Creating ex4.py