Added return for models