cleaned up shader code