Code revue TODO fixes