final changes to make the changes in plots work