fix bugs
Merge pull request #9 from ArthurStam/bumblebee

fix bugs