moving things to register_class - the correct place to be