Summary:
aadding some changes for optional args

Adding range and kwargs support

Test Plan: n/a

Reviewers: michelle, zasgar, #engineering, oazizi

Reviewed By: #engineering, oazizi

Subscribers: oazizi

Differential Revision: https://phab.corp.pixielabs.ai/D434

GitOrigin-RevId: e6acc2efdd07ab8e8c88e900c253d16f54b30c8c