Copy code from asst 3 to asst 4