Minor changes & TODOs