Added class to wrapper