custom input for states
Fix simple_for usage errors