Tweaks and fixes.
A small todo.