Minor modifications to plots