removed debug
Updated training model