Add trap function