Minor formatting issue, adding todo comment