added init for training
Added init file