To do: 1. Add ray tracing
To do: