adding plot_test and updates