Blocking .wait() should be optional

- also removed some unneeded imports

can we also update the pypi?