Add cs171 code