Added base class for TF testing + added TokenStream as a param