Add code for day 3