minor changes in my research code