Changed button example code for new update