Add try/except for errors