Update optical_system.py