The code for the button is now based on switch/case instead of if/else. Would allow for easier extention.

Signed-off-by: Sijmen <sijmen.hoogendijk@gmail.com>