Updated register ui