[UPDATE] switch instead of if