clean up + todo