Add optical medium

**To Do**:
- [x] Initialize medium with refractive index(Better API?)
- [x] Add tests for medium

Though that's unrelated to this PR.