Handle yellow lights 

Fixes #8.

Can you review this, @camdez?