Some changes to the wolf code