Adding hw6 - same prob though