Improve Bert Text Generation