Switched code to week4 day 1