Keep model & input together