added stuff to make plots