Improved uncertainty quantification for neural networks with Bayesian last layer