Refactor: Handle all enum cases for trig_fn