added separate test class