Working with Lee to fix bug.
adding migration
Fixed with Lee and fixed button