added a todo
moved code to lights.go file