Minor tweaks
Add TODO