Add TODO flags and clean whitespace