Simplify the get_input_type method, add multi type