Fixed minor TODOs