Fixed missing TODOs