Add function to buttons on info page