Extracted module commited, both training and testing part.