Update button.js

Code is cleaner