Feature/setting up models