Add input_spec and output_spec