add new class for persistance