added a couple necessary classes