this is work from day 4 of class