Choosing a limited set of words/phrases/phrasal verbs and writing an essay based on them is also a good way of training.