Using a counter in a python function