Add rule for zip+count optimization