Fix documentation typo
Add arbitrary, user-specifiable heat loss function