add screen shut function