Instead of using a `match` this can use `map`