TODO stuff in main