Clean up js

- Retain meshes in a 2d array
- Use the lambertian material