Updated python so we can use lists instead of dicts