implement changes from feedback to day 2 ex3