add logout functionality after login, and other small thing. Feel free to discover.