Add plotting routines and plots