Allow passing flags

Fixes #5