Main Code

Just need to add the if statements