Add regularization / pretty prints