updated code to reflect changes #251

Closed corinne-riddell closed 7 years ago