Clean code
Add TODO