Setting up Unit of work pattern