Don't forget this last comment! Those two lines can replace the big `if` altogether.