added functions to class