Small fixes. Mind the TODO.