added necessary class