update with new code and plots