TODO cleanups