fix grid size and wrap main() call