Wrote a quick and dirty test class