Adjust plot saving