Clean up files, remove TODO item