Put the new code under try-catch