clean up & add todo