Cleanup + TODOs