Use two-argument form of next() and use a return-statement instead of an explicit raise StopIteration