added stuff for week-4