Update and rename 21.py to main.py