Added class User to models