Updated methods so class compiles