tidy up TODO