small improvement
Buttons working!