Added stuff to button click thing