Added function for setting model for yellow balloon