attempted bug fix / more plot setup