Added new class to distribution model