Added/fixed a couple tests for the features just added