Fix for #4 -- allow callbacks in `run()`