Modified array_problems

Nice solution!