Added docstring to wait_for

Added full docstring for wait module