added wrapper class