Main class, updated trining data