Seperated lapiz mock from demo code