figuring out python arrays