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

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