Quick change to how classes are registered