Added ability to modify a button