add updated wrapper