Replace klass with described_class