bugfix updates for over fitting