change color generator
minor changes
training files