Added training data and updated training module