Updated movement tests to use new Map structure.
- also updated some docstrings, and added missing docstrings