Streamlined update_state usage

Switched from kwarg usage to arg