Move all TODO statements to the main class.