Add try catch here