Initial submission task1.py