Replace magic number 4 with NUM_STATES. Add comments.

Replace magic number 4 with NUM_STATES. Add comments. #141