Added column argument for sparsityHeatMap