import Data.Maybe (mapMaybe)
import Data.List (elemIndex)