updated unit tests for additional functionality