Updated css and fixed DI