added setter for Entity and DTO and some code around that