Update setClasses code style