Same as above. Remove the map and use a simple switch statement.