Add spatial component in representation
Add rewrite as exp function
Add more tests