Added JS to deal with remove button click events