Add reference to example script in vae/__init__.py