Create task2_final.py