Refactor parameter types