Add inputs to do forward run