Added get_rows and get_cols method