Add tracking of button clicks