add flatten helper function
2 files changed