Clean up code:

Add function set_unicast_flow
Merge pull request #39 from powerist/working

Clean up code: