Fixed the class name to correspond correctly to the README file, and fixed an 'undefined variable' problem