Improve login register