Added entity framework