adding wrapper class