Improve mafia.

Bug fixes.