More error fixing

Fixing errors in plot