diff --git a/src/ray.zig b/src/ray.zig
index 5e8a6dd..91b39a4 100644
--- a/src/ray.zig
+++ b/src/ray.zig
@@ -12,21 +12,32 @@ pub fn init(origin: zm.Vec, direction: zm.Vec) Ray {
     };
 }
 
-pub fn color(self: *Ray) zm.Vec {
-    if (hit_sphere(zm.f32x4(0, 0, -1, 0), 0.5, self)) {
-        return zm.f32x4(1, 0, 0, 1);
+pub fn at(self: *Ray, t: f32) zm.Vec {
+    return self.orig + zm.f32x4s(t) * self.dir;
+}
+
+pub fn color(r: *Ray) zm.Vec {
+    const t = hitSphere(zm.f32x4(0, 0, -1, 0), 0.5, r);
+    if (t > 0.0) {
+        const N = zm.normalize3(r.at(t) - zm.f32x4(0, 0, -1, 0));
+        return zm.f32x4s(0.5) * zm.f32x4(N[0] + 1, N[1] + 1, N[2] + 1, 1);
     }
 
-    const unit_direction = zm.normalize3(self.dir);
+    const unit_direction = zm.normalize3(r.dir);
     const a = 0.5 * (unit_direction[1] + 1.0);
     return zm.f32x4s(1.0 - a) * zm.f32x4s(1.0) + zm.f32x4s(a) * zm.f32x4(0.5, 0.7, 1.0, 1.0);
 }
 
-fn hit_sphere(center: zm.Vec, radius: f32, r: *Ray) bool {
+fn hitSphere(center: zm.Vec, radius: f32, r: *Ray) f32 {
     const oc = r.orig - center;
-    const a = zm.dot3(r.dir, r.dir)[0];
-    const b = 2.0 * zm.dot3(oc, r.dir)[0];
+    const a = zm.lengthSq3(r.dir)[0];
+    const half_b = zm.dot3(oc, r.dir)[0];
     const c = zm.dot3(oc, oc)[0] - radius * radius;
-    const discriminant = b * b - 4 * a * c;
-    return discriminant >= 0;
+    const discriminant = half_b * half_b - a * c;
+
+    if (discriminant < 0) {
+        return -1.0;
+    } else {
+        return (-half_b - @sqrt(discriminant)) / a;
+    }
 }