completed code from day1 week 5