Fixed training data and added mistakenly deleted line