Add the sleep feature