Added button functionality. Buttons working as intended.