Starting work on the button implementation