Added getTokenCount back in.