Add setId method