Build your first transformer using BERT framework 2