Updating JS for new behaviors