added user class in models.py, changed main file accordingly