Add comments and docstrings to tests
Add test for optional/default arguments