User Input Array Size Python Coder