Add functionality so that the buttons work, plus restyle some stuff.