Add classes for register