Modified some parts and added DbSet properties to context