Updated previous commit, ivar needs to be added to the class extension