Add day 3 code