Why not use a couple of ifs + return? I think switch is a bit unreadable,