Initial working demo for DI