Create Assignment 3 code