Adding wait helper