Clean up test
Merge pull request #44 from joneslee85/clean-up-test

Clean up test