added module for plotting the results