Adding Wait() method