Adding a couple of class/header files that were missing from the project.