Remove unit of work and create in repository, add unit test class and some images for later