Updating test class