Add missing paramaters
add build_grid method to customize grid