Update the comments on states to reflect up-to-date algorithm