Added Test class for easy testing