added training part