update to work with 4.0.0
4.0.0 update
Merge pull request #1 from dapil/master

Master