Fix demos to accommodate new api changes.
Merge pull request #271 from cowlicks/fix-demos

Fix demos to accommodate new API changes.