Adding extra wait