Add set_light method
Add tests for set_light