Created a new test class and improved the getType method