Přitom to tvoje řešení mi přišlo jako příliš komplikované, stačí přidat jeden parametr do funkce a řídit se jím.