Change class names to fit my config