Implemented DTO

- implemented DTO + tests