Update js for new markup