Updated the states for the math trick example.