Snell's law and total internal reflection in code | StemPy Lab