Added test
Updated test