added feature with a button, button works only