Add delay functionality