Improving Stateful Premise Selection with Transformers