Fixed import sys issue
Added update method