Rename state_dict.py to states.py