Update TODO
Add another parameter