Added lightning generator and basic draw tests for that