Reworked EF to allow for migrations