Add new class for YML