optimization code(add new file)