Add mock demos
Revise basic mock demo
Add pytest