Would it be better style to call a function doing the calculation or is the Switch statement fine?
Neither? switch statements are generally preferred when there are 4 or more cases. With only two cases, if statements are faster. Of course, that just means that you get to the delay() sooner, so, never mind.