Add support for Lightning