Training mode added