Revamped module, added a demo script