Variable Length Array Python