Fix some todos
merge TODO fixes