Registration UI revamp