code for cs344