Updated to work with 2D lighting.