re working cs445 hw