Add rudimentary tests for issue #1
Add rudimentary implementation for custom attributes key - closes issue #1