Switched from ls to os.listdir