Add $schema key to the generated schema