just made it  case statement