added the python code used to create figures