Added "Mafia" functionality

Mafia section now works as well