TODOs + cleanups