Add functionality to some buttons