tidying up TODO