Added power button mapping

Also changed from if-else to a dictionary