Simplify method. Fix bug.
One line.
Use array_map.