cs228: more hw.