Added clean_state_files
Proper naming
Added docs