Small updates to plotting scripts.