Change from count() to size()