Stony brook CSE 214 homework 5