Add signal handling to compute