improve upgrade
Merge pull request #300 from ahilles107/update_version

improve upgrade