Taking Input Array In Python Problem