Minor changes to plots