Assistant coach Shaun Edwards expects Wales to put in a much-improved display in Saturday's second Test against Japan in Tokyo.