Minor changes to DTO