Messing with buttons, but has a bug :<