added stuff for section week 4