Added class in functional test