Clean up the rest of the html and add classes.