Add comments
Update TODOs