Added more to test file. Fixes #3