Adding some test classes to see it in action