Change code to add refraction effect