Add tests for optional arguments