Minor tweak to file splitting script