improve ufo code