Added import error try/catch block