Added code first classes