Added TODO
Fixed lemma