Make input dir a parameter