Fix import #1
Add a custom exception