you should not leave your implementation the same just because thats how you've written it the first time
if you're thinking of implementing something - think of the "useful", underlying "business" logic first, not in python terms