Add update function
Merge pull request #1 from pukka/test

Add update function