Add class for exportation to python. Correct results