Add refresh_func parameter to wait_for()