２で作ったコードとほぼ同じだったので割とすんなりできた