Add key selector argument to map()