Add signal handling support