Added import of exception class