Added functionality to get columns and rows