Add demonstration tests to match documentation