Add basic signal handling