Minor changes to study