Add some JS and buttons