Add a couple class files