Define Size Of Array In Python Problems On Loops