add todo
tidy up