Small fixes, TODO update