Adds doctests and removes terrible demo.