Add argument to migrations