Updating test, not complete