add stay awake feature