added new class extension code