Add tests for arguments