changes to plot functions, cleaned arguments, made pretty