hw6 done but without tests code