add - check for pch

By using `return` here and below, you break out of the `map` and return only a single element and not an array anymore, that's the cause why the tests fail. Yep, this is Ruby. Just leave out the return and as everything is an expression it will then just work. Furthermore `flatten` doesn't sort out empty strings, so in this place, you would need to evaluate to `nil` instead.