Add a try-catch