Add finish tests for prototype