Added args to swap_grid method (for button, etc).