Add addition catch block for exception too