update code to test
Merge pull request #338 from katiewood/master

update code to test