Add plots for potential_test, cosmetic changes