Refactoring. Added login/register to mainscreen