Update to EF and enabling update to related entities