Add buttons to choose grid dimensions, close #2