Added new unit test cases + fixed test (associative array didn't work correctly)