almost solve 3.2