Added entity framework stuff