Minor changes for final result plot