Clean up class names in the partial