changed if else to switch