Update main.cpp

Don't need to test for both conditions since "else if" is being used
Merge pull request #3 from ColorfulGecko/patch-3

Update main.cpp