Starting to move level code from main.py to levels.py