correctly add the class to the pre/code tags