Added the nbdensitytraining.py