Refactor
Add DTO to test Swagger