Re-write plot functions.