Working on the button script.