multiply counts by lambda as well, make sure we properly print our trained matrix