Working on second part of registration UI