Added in a TODO

too sleepy to do now. But didn't want to forget it later