added header and new class