Fixing TODOs