new version of last figure + python code