Added Wake Lock feature