refactor

make functions to give names to and make distinct different sections
use experiment_utils.plot_results to save plots