Small modifications to SA source term to look cleaner