Assignment 1 for CS344