more cleanup & todos