Added entropy regularization