Simple version for the demo