add Entity and MockFactory class