added some more js and a button