Cleaned up the if statements in main.js