- Add option to keep screen on
- Improve timing, yet it is not perfect
