Updated migrations to reflect association