Edits to update the code with respect to the TL code