Add functionality to class version