more readability
dictionary instead of switch