Update js for buttons