Added training script
Working on the category graph