Solved this in a oneliner (if you forget the number array)