Adding classes to XML output