Added lightning effect function