Add try catch