Adding login functionality and other cool features