add hw2
python realisation