Add scripts for test and plot