maybe you want to use a swich here?