cs228: hw