handling case of dict key as integer + adding some more tests
adding more detailed documentation