Refactor to allow for explicit solving of exponential decay
Alternate state class