Improving a bit the python part (unfinished).