going to use python to generate figure