Improve the code using [Flow](https://flow.org/en/docs/install/)
Improve the code using Flow