added lights code. Need to test now