Add `timer_wrapper` functionality