Update design for pre-register