Updated another demo to using the recently changed plotting function.