some minor changes in my research code