Changed some button functionality.