Reduce Array Size Python Coderhouse Mxa