added code for bubble plots