Initialzie method added
Oops. Forgot to add models