Define Size Of Array In Python Problem