Temp sample

meh.py

def pop(initial_state, time):
    state_cnt = [initial_state.count(i) for i in range(0, 9)]
    for _ in range(0, time):
        state_cnt = state_cnt[1:] + state_cnt[:1]
        state_cnt[6] += state_cnt[8]
    return sum(state_cnt) 

def partOne(data):
    return pop(data, 80)

def partTwo(data):
    return pop(data, 256)