Added a new case for the switch statement to try and make it a little
more readible.