cleanup + add todos