more cleanup + todos