Forgot to add function to class