Add parameter flexibility and update documentation