adding count hack for display