Add Flow class
Add test