Messing around with adding buttons, extra functionality