Add TODO. Some formatting