Here is the in class addition to the code