Button updates

Updated js for buttons