Add migrations

@stephtzhang @aferragano please review and merge