Add tests
Create Map wrapper