Update challengeMain.py