Modify Keep Screen On code in Stage

* Use addFlags() instead of setFlags()
* Remove hardcoded int in tests

Tested on device with 15s screen timeout. Works.