Update to login and register layouts