Python implementation to solve the zebra puzzle problem