[ML] Remove unneccessary comments and already resolved TODOs