add wait method