Added migration for locations.
Added changes to make tests pass.