Missing a "fi" in added code, oops!