Improve the research code.