1. change module name
2. add flatten dict