Poprawiony UnityOfWork i encje DB