Readd print statements, not sure if plotting is working correctly.