Change to have a user and friend class