Added DI demo
cleanup