Updating js for new behaviors