Verified it works:). Just need to test live training.