Modifications to add DTOs to the application