fix: fix return type hint for grid(), add a few others (#94)

fix: fix return type hint for grid(), add a few others (#94) #91