Add missing imports
Merge pull request #235 from a5anka/flow_control

Add missing imports