Added setter function

Example code