Updated initial migration to work with custom user models