Set up the switch (if-else)