Added test for functionality.