Added tests to ensure example passes