K o z í  H r á d e k, rekonstrukce hradu