Created class for lightning, doesn't work