Code First Migration, and some other changes. Check the code! :P