Added TODO to main.lua

These are cleanup and refactor things I should do.