[Dev] Refactored code and added more regexes to match training messages.