putting code in right places and login screen styling