minor cleanup - add todos