Make JS aware of new HTML structure