Better power button management