Wrote `Map` and `Remove` using list comprehensions