add training flows + fix timeouts