added unit of work to solution