added week4 quick code