Adding set function