Reviewec code, added/removed some TODOs