Fixed button highlighting on click