updated wrapper class