add code for day 4