Remove reflection
Update TODOs