Update main.cpp

Don't need to test for both conditions since "else if" is being used