Added code for night vision effect