Add some flow stuff