Added controls for register