Add code for set methods