Added tests to demonstrate usage