small changes - TODOs