Changed to call plot functions on source