Make test automated
Merge pull request #2 from stephenmathieson/automated-tests

Make test automated