Balloon check update

Restructure balloon check case. Make it more clear and flexible. Also add a case about it.