Add todo fixes
Add todo