got the set method working