Added training (works like production) and refactored