Add flags and split TODO.