small fix. updated todo