Improve code
Add login functionnality