stoch penalty: this one works