Driver improved to work with the AugmentedGrammar class.