Update ui
Add a method to set grid size