added small plotting of results