Defining Array Size In Python