Some new fixes and new login / register screen