Add finish() call