minor changes in test main and plot script