Set up if else statements in the main.py page