You could combine this if/else statement with the above one if you want- not sure if it's clearer/simpler. Up to you.