Added models and mongoose setup