Added in balloon functionality