Feature/add input