Update test to include more testing.