Fixing project01.py