Minor changes

TODOs