Implementing #24 - added upgrade functionality.
Implementing #24