Forgot to update the User class as well
Added a vote class