Add set_counter function