Replace magic number 4 with NUM_STATES. Add comments. (#108)

Replace magic number 4 with NUM_STATES. Add comments. (#108) #146