added day 2 code