minor fixes and todo.