Warmup: fixed another math error, added plotting of values vs state.