adding count output fix