Tweak to button example