add set function
Merge pull request #2 from Doomsta/feature/setFunction

Feature/set function