Add a try / catch