Improve Demo Module (2)