Wrapped up most TODOs