changed class for wrapper