Add sleep functionality