Change state actions 251

Fixes #251 
This dictionary seems unnecessary? You could just to `to_stop = set([unit1])`.