Write TODO and organize the program