add code for three