main: Remove unused import
main: Move TODO