Updated mse tracking code
Re-removed pygments