Add parameter handling